"""
@FileName：2判断文件后缀名.py
@Author：lmz
@Time：2021/5/20 22:00
2、遍历一个文件夹，包括自文件夹，找到里面所有bmp后缀的文件，
判断文件是否完整，不完整的文件在文件名后加.broken的后缀。
"""

import os
import struct

YourFile_path='hello'
files = os.listdir(YourFile_path)
# files = [Video_path + f for f in files if f.endswith(('.mp4','.avi'))]

files = [f for f in files if f.endswith(('.bmp'))]
print(files)
f = open("save1.bmp", 'rb')
bmp_header_b = f.read(0x36)
print(bmp_header_b)
print('')

bmp_header_s = struct.unpack('<2sI2H4I2H6I', bmp_header_b)
print(bmp_header_s)
print('')